import { login, logout, getInfo } from '@/api/user'
import { getToken, setToken, removeToken } from '@/utils/auth'
import { resetRouter } from '@/router'

const getDefaultState = () => {
  return {
    token: getToken(),
    name: '',
    avatar: '',
    user: null,  // 添加用户信息字段,
    roles: []  // 添加角色数组
  }
}

const state = getDefaultState()

const mutations = {
  RESET_STATE: (state) => {
    Object.assign(state, getDefaultState())
  },
  SET_TOKEN: (state, token) => {
    state.token = token
  },
  SET_NAME: (state, name) => {
    state.name = name
  },
  SET_AVATAR: (state, avatar) => {
    state.avatar = avatar
  },
  SET_USER: (state, user) => {
    state.user = user
    // 根据用户信息设置角色
    if (user.studentId === '000001') {
      state.roles = ['admin']
    } else if (user.identity === '1') {
      state.roles = ['teacher']
    } else {
      state.roles = ['student']
    }
  }
}

const getters = {
  roles: state => state.roles || [],
  user: state => state.user || {}
}

const actions = {
  // user login
  login({ commit }, userInfo) {
    const { studentId, password } = userInfo
    return new Promise((resolve, reject) => {
      login({ studentId: studentId.trim(), password: password }).then(response => {
        const { data } = response
        commit('SET_TOKEN', data.token)
        commit('SET_USER', data)  // 保存用户信息
        console.log(data,'data');
        localStorage.setItem('userInfo', JSON.stringify(data.user))
        setToken(data.token)
        resolve()
      }).catch(error => {
        console.log(11)
        commit('SET_TOKEN', '1111')
        setToken('1111')
        reject(error)
      })
    })
  },

  // get user info
  getInfo({ commit, state }) {
    return new Promise((resolve, reject) => {
      getInfo(state.token).then(response => {
        const { data } = response

        if (!data) {
          return reject('Verification failed, please Login again.')
        }
        const { name, avatar } = data
        commit('SET_NAME', name)
        commit('SET_AVATAR', avatar)
        commit('SET_USER', data)  // 保存用户信息
        resolve(data)
      }).catch(error => {
        reject(error)
      })
    })
  },

  // user logout
  logout({ commit, state }) {
    return new Promise((resolve, reject) => {
      removeToken() // must remove  token  first
      resetRouter()
      commit('RESET_STATE')
      resolve()
    })
  },

  // remove token
  resetToken({ commit }) {
    return new Promise(resolve => {
      removeToken() // must remove  token  first
      commit('RESET_STATE')
      resolve()
    })
  }
}

export default {
  namespaced: true,
  state,
  getters,
  mutations,
  actions
}

